/*
 * Copyright (C) 2008 Emweb bv, Herent, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#include <boost/algorithm/string.hpp>
#include <iostream>
#include <cctype>
#include <exception>

#include "Wt/WApplication.h"
#include "Wt/WContainerWidget.h"
#include "Wt/WLogger.h"
#include "Wt/WTemplate.h"

#include "EscapeOStream.h"
#include "WebUtils.h"
#include "DomElement.h"
#include "RefEncoder.h"
#include "WebSession.h"

namespace Wt {
LOGGER("WTemplate");

bool WTemplate::_tr(const std::vector<WString>& args,
                    std::ostream& result)
{
  if (args.size() >= 1) {
    WString s = WString::tr(args[0].toUTF8());
    for (unsigned j = 1; j < args.size(); ++j)
      s.arg(args[j]);
    result << s.toXhtmlUTF8();
    return true;
  } else {
    LOG_ERROR("Functions::tr(): expects at least one argument");
    return false;
  }
}

bool WTemplate::_block(const std::vector<WString>& args,
                              std::ostream& result)
{
  if (args.size() < 1)
    return false;

  WString tblock = WString::tr(args[0].toUTF8());
  for (unsigned i = 1; i < args.size(); ++i)
    tblock.arg(args[i]);

  this->renderTemplateText(result, tblock);

  return true;
}

bool WTemplate::_while(const std::vector<WString>& args,
                              std::ostream& result)
{
  if (args.size() < 2)
    return false;

  WString tblock = WString::tr(args[1].toUTF8());
  for (unsigned i = 2; i < args.size(); ++i)
    tblock.arg(args[i]);

  while (conditionValue(args[0].toUTF8()))
    this->renderTemplateText(result, tblock);

  return true;
}

bool WTemplate::_id(const std::vector<WString>& args,
                    std::ostream& result)
{
  if (args.size() == 1) {
    WWidget *w = this->resolveWidget(args[0].toUTF8());
    if (w) {
      result << w->id();
      return true;
    } else
      return false;
  } else {
    LOG_ERROR("Functions::tr(): expects exactly one argument");
    return false;
  }
}

#ifndef WT_TARGET_JAVA
bool WTemplate::Functions::tr(WTemplate *t, const std::vector<WString>& args,
                              std::ostream& result)
{
  return t->_tr(args, result);
}

bool WTemplate::Functions::block(WTemplate *t, const std::vector<WString>& args,
                              std::ostream& result)
{
  return t->_block(args, result);
}

bool WTemplate::Functions::while_f(WTemplate *t, const std::vector<WString>& args,
                              std::ostream& result)
{
  return t->_while(args, result);
}

bool WTemplate::Functions::id(WTemplate *t, const std::vector<WString>& args,
                              std::ostream& result)
{
  return t->_id(args, result);
}

#else

bool WTemplate::TrFunction::evaluate(WTemplate *t,
                                     const std::vector<WString>& args,
                                     std::ostream& result) const
{
  try {
    return t->_tr(args, result);
  } catch (std::io_exception ioe) {
    return false;
  }
}

bool WTemplate::BlockFunction::evaluate(WTemplate *t,
                                     const std::vector<WString>& args,
                                     std::ostream& result) const
{
  try {
    return t->_block(args, result);
  } catch (std::io_exception ioe) {
    return false;
  }
}

bool WTemplate::WhileFunction::evaluate(WTemplate *t,
                                    const std::vector<WString>& args,
                                    std::ostream& result) const
{
  try {
    return t->_while(args, result);
  } catch (std::io_exception ioe) {
    return false;
  }
}

bool WTemplate::IdFunction::evaluate(WTemplate *t,
                                     const std::vector<WString>& args,
                                     std::ostream& result) const
{
  try {
  return t->_id(args, result);
  } catch (std::io_exception ioe) {
    return false;
  }
}

#endif

WTemplate::WTemplate()
  : previouslyRendered_(nullptr),
    newlyRendered_(nullptr),
    encodeInternalPaths_(false),
    encodeTemplateText_(true),
    changed_(false),
    widgetIdMode_(TemplateWidgetIdMode::None)
{
  plainTextNewLineEscStream_ = new EscapeOStream();
  plainTextNewLineEscStream_->pushEscape(EscapeOStream::PlainTextNewLines);
  setInline(false);
  setTemplateText(WString::Empty);
}

WTemplate::WTemplate(const WString& text)
  : previouslyRendered_(nullptr),
    newlyRendered_(nullptr),
    encodeInternalPaths_(false),
    encodeTemplateText_(true),
    changed_(false),
    widgetIdMode_(TemplateWidgetIdMode::None)
{
  plainTextNewLineEscStream_ = new EscapeOStream();
  plainTextNewLineEscStream_->pushEscape(EscapeOStream::PlainTextNewLines);
  setInline(false);
  setTemplateText(text);
}

WTemplate::~WTemplate()
{
  clear();
  delete plainTextNewLineEscStream_;
}

void WTemplate::clear()
{
  // Widgets should be orphaned before they are deleted, because
  // when WWebWidget calls removeFromParent(), the parent should be null.
  for (WidgetMap::iterator it = widgets_.begin(); it != widgets_.end(); ++it) {
    WWidget *w = it->second.get();
    if (w)
      widgetRemoved(w, false);
  }

  widgets_.clear();

  strings_.clear();
  conditions_.clear();

  changed_ = true;
  repaint(RepaintFlag::SizeAffected);
}

#ifndef WT_TARGET_JAVA
void WTemplate::addFunction(const std::string& name, const Function& function)
{
  functions_[name] = function;
}
#else
void WTemplate::addFunction(const std::string& name, const Function *function)
{
  functions_[name] = *function;
}
#endif

void WTemplate::setCondition(const std::string& name, bool value)
{
  if (conditionValue(name) != value) {
    if (value)
      conditions_.insert(name);
    else
      conditions_.erase(name);

    changed_ = true;
    repaint(RepaintFlag::SizeAffected);
  }
}

bool WTemplate::conditionValue(const std::string& name) const
{
  return conditions_.find(name) != conditions_.end();
}

std::unique_ptr<WWidget> WTemplate::removeWidget(WWidget *widget)
{
  const std::string *k = Utils::keyForUniquePtrValue(widgets_, widget);
  if (k)
    return removeWidget(*k);
  else
    return std::unique_ptr<WWidget>();
}

void WTemplate::iterateChildren(const HandleWidgetMethod& method) const
{
  for (WidgetMap::const_iterator it = widgets_.begin(); it != widgets_.end(); ++it) {
    WWidget *w = it->second.get();
    if (w)
#ifndef WT_TARGET_JAVA
      method(w);
#else
      method.handle(w);
#endif
  }
}

#ifndef WT_TARGET_JAVA
void
#else // WT_TARGET_JAVA
WWidget*
#endif // WT_TARGET_JAVA
WTemplate::bindWidget(const std::string& varName,
                      std::unique_ptr<WWidget> widget)
{
  bool setNull = !widget;
#ifdef WT_TARGET_JAVA
  WWidget *widgetPtrCopy = widget.get();
#endif // WT_TARGET_JAVA

  if (!setNull) {
    strings_.erase(varName);

    switch (widgetIdMode_) {
    case TemplateWidgetIdMode::None:
      break;
    case TemplateWidgetIdMode::SetObjectName:
      widget->setObjectName(varName);
      break;
    case TemplateWidgetIdMode::SetId:
      widget->setId(varName);
    }
  } else {
    StringMap::const_iterator j = strings_.find(varName);
    if (j != strings_.end() && j->second.empty()) {
#ifndef WT_TARGET_JAVA
      return;
#else // WT_TARGET_JAVA
      return widgetPtrCopy;
#endif // WT_TARGET_JAVA
    }
    strings_[varName] = WString();
  }

  removeWidget(varName);
  manageWidget(widgets_[varName], std::move(widget));

  changed_ = true;
  repaint(RepaintFlag::SizeAffected);

#ifdef WT_TARGET_JAVA
  return widgetPtrCopy;
#endif // WT_TARGET_JAVA
}

std::unique_ptr<WWidget> WTemplate::removeWidget(const std::string& varName)
{
  std::unique_ptr<WWidget> result;

  WidgetMap::iterator i = widgets_.find(varName);
  if (i != widgets_.end()) {
    result = manageWidget(i->second, std::unique_ptr<WWidget>());
#ifndef WT_TARGET_JAVA
    widgets_.erase(i);
#else
    widgets_.erase(varName);
#endif

    changed_ = true;
    repaint(RepaintFlag::SizeAffected);
  }

  return result;
}

void WTemplate::setWidgetIdMode(TemplateWidgetIdMode mode)
{
  widgetIdMode_ = mode;
}

void WTemplate::bindEmpty(const std::string& varName)
{
  bindWidget(varName, nullptr);
}

void WTemplate::bindString(const std::string& varName, const WString& value,
                           TextFormat textFormat)
{
  WWidget *w = resolveWidget(varName);
  if (w)
    bindWidget(varName, nullptr);

  WString v = value;

  if (textFormat == TextFormat::XHTML && v.literal()) {
    if (!removeScript(v))
      v = escapeText(v, true);
  } else if (textFormat == TextFormat::Plain)
    v = escapeText(v, true);

  StringMap::const_iterator i = strings_.find(varName);

  if (i == strings_.end() || i->second != v) {
    strings_[varName] = v;

    changed_ = true;
    repaint(RepaintFlag::SizeAffected);
  }
}

void WTemplate::bindInt(const std::string& varName, int value)
{
  bindString(varName, std::to_string(value), TextFormat::UnsafeXHTML);
}

bool WTemplate::resolveFunction(const std::string& name,
                                const std::vector<WString>& args,
                                std::ostream& result)
{
  FunctionMap::const_iterator i = functions_.find(name);

  if (i != functions_.end()) {
#ifndef WT_TARGET_JAVA
    bool ok = i->second(this, args, result);
#else
    bool ok = i->second.evaluate(this, args, result);
#endif // WT_TARGET_JAVA

    if (!ok)
      result << "??" << name << ":??";

    return true;
  }

  return false;
}

void WTemplate::resolveString(const std::string& varName,
                              const std::vector<WString>& args,
                              std::ostream& result)
{
  /*
   * FIXME: have an extra result parameter which indicates whether the
   * widget is view-only. Better to do that in resolveValue() and
   * provide a utility method that converts a widget to XHTML ?
   */

  StringMap::const_iterator i = strings_.find(varName);
  if (i != strings_.end())
    result << i->second.toUTF8();
  else {
    WWidget *w = resolveWidget(varName);
    if (w) {
      w->setParentWidget(this);

      if (previouslyRendered_
          && previouslyRendered_->find(w) != previouslyRendered_->end()) {
        result << "<span id=\"" << w->id() << "\"> </span>";
      } else {
        applyArguments(w, args);
        w->htmlText(result);
      }

      if (newlyRendered_)
        newlyRendered_->push_back(w);
    } else
      handleUnresolvedVariable(varName, args, result);
  }
}

void WTemplate::applyArguments(WWidget *w, const std::vector<WString>& args)
{
  for (unsigned i = 0; i < args.size(); ++i) {
    std::string s = args[i].toUTF8();
    if (boost::starts_with(s, "class="))
      w->addStyleClass(WString::fromUTF8(s.substr(6)));
  }
}

void WTemplate::handleUnresolvedVariable(const std::string& varName,
                                         WT_MAYBE_UNUSED const std::vector<WString>& args,
                                         std::ostream& result)
{
  result << "??" << varName << "??";
}

WWidget *WTemplate::resolveWidget(const std::string& varName)
{
  WidgetMap::const_iterator j = widgets_.find(varName);
  if (j != widgets_.end())
    return j->second.get();
  else
    return nullptr;
}

const WString& WTemplate::resolveStringValue(const std::string& varName)
{
  StringMap::const_iterator i = strings_.find(varName);
  if (i != strings_.end())
    return i->second;
  else
    return WString::Empty;
}

std::vector<WWidget *> WTemplate::widgets() const
{
  std::vector<WWidget *> result;

  for (WidgetMap::const_iterator j = widgets_.begin();
       j != widgets_.end(); ++j)
    result.push_back(j->second.get());

  return result;
}

std::string WTemplate::varName(WWidget *w) const
{
  for (WidgetMap::const_iterator j = widgets_.begin();
       j != widgets_.end(); ++j)
    if (j->second.get() == w)
      return j->first;

  return std::string();
}

void WTemplate::setTemplateText(const WString& text, TextFormat textFormat)
{
  text_ = text;

  if (textFormat == TextFormat::XHTML && text_.literal()) {
    if (!removeScript(text_))
      text_ = escapeText(text_, true);
  } else if (textFormat == TextFormat::Plain)
    text_ = escapeText(text_, true);

  changed_ = true;
  repaint(RepaintFlag::SizeAffected);
}

void WTemplate::updateDom(DomElement& element, bool all)
{
  if (changed_ || all) {
    std::set<WWidget *> previouslyRendered;
    std::vector<WWidget *> newlyRendered;

    for (WidgetMap::const_iterator i = widgets_.begin(); i != widgets_.end();
         ++i) {
      WWidget *w = i->second.get();
      if (w && w->isRendered()) {
        if (w->webWidget()->domCanBeSaved()) {
          previouslyRendered.insert(w);
        } else {
          unrenderWidget(w, element);
        }
      }
    }

    bool saveWidgets = element.mode() == DomElement::Mode::Update;

    previouslyRendered_ = saveWidgets ? &previouslyRendered : nullptr;
    newlyRendered_ = &newlyRendered;

    std::stringstream html;
    renderTemplate(html);

    previouslyRendered_ = nullptr;
    newlyRendered_ = nullptr;

    for (unsigned i = 0; i < newlyRendered.size(); ++i) {
      WWidget *w = newlyRendered[i];
      if (previouslyRendered.find(w) != previouslyRendered.end()) {
        if (saveWidgets)
          element.saveChild(w->id());
        previouslyRendered.erase(w);
      }
    }

    if (encodeTemplateText_)
      element.setProperty(Property::InnerHTML, html.str());
    else
      element.setProperty(Property::InnerHTML, encode(html.str()));

    for (std::set<WWidget *>::const_iterator i = previouslyRendered.begin();
         i != previouslyRendered.end(); ++i) {
      WWidget *w = *i;
      // it could be that the widget was removed/deleted in the mean time
      // as a side-effect of rendering some of the widgets; thus we check
      // that the widget is still a child
      for (WidgetMap::const_iterator j = widgets_.begin();
           j != widgets_.end(); ++j) {
        if (j->second.get() == w) {
          unrenderWidget(w, element);
          break;
        }
      }
    }

    WApplication::instance()->session()->renderer()
      .updateFormObjects(this, true);

    changed_ = false;
  }

  WInteractWidget::updateDom(element, all);
}

void WTemplate::unrenderWidget(WWidget *w, DomElement &el)
{
  std::string removeJs = w->renderRemoveJs(false);
  if (removeJs[0] == '_')
    el.callJavaScript(WT_CLASS ".remove('" + removeJs.substr(1) + "');", true);
  else
    el.callJavaScript(removeJs, true);
  w->webWidget()->setRendered(false);
}

std::string WTemplate::encode(const std::string& text) const
{
  WApplication *app = WApplication::instance();

  if (app && (encodeInternalPaths_ || app->session()->hasSessionIdInUrl())) {
    WFlags<RefEncoderOption> options;
    if (encodeInternalPaths_)
      options |= EncodeInternalPaths;
    if (app->session()->hasSessionIdInUrl())
      options |= EncodeRedirectTrampoline;
    return EncodeRefs(WString::fromUTF8(text), options).toUTF8();
  } else
    return text;
}

void WTemplate::renderTemplate(std::ostream& result)
{
  renderTemplateText(result, templateText());
}

bool WTemplate::renderTemplateText(std::ostream& result, const WString& templateText)
{
  errorText_ = "";

  std::string text;
#ifndef WT_TARGET_JAVA
  if (encodeTemplateText_)
    text = encode(templateText.toXhtmlUTF8());
  else
    text = templateText.toXhtmlUTF8();
#else // WT_TARGET_JAVA
  if (encodeTemplateText_)
    text = encode(WString(templateText).toXhtmlUTF8());
  else
    text = WString(templateText).toXhtmlUTF8();
#endif

  std::size_t lastPos = 0;
  std::vector<WString> args;
  std::vector<std::string> conditions;
  int suppressing = 0;

  for (std::size_t pos = text.find('$'); pos != std::string::npos;
       pos = text.find('$', pos)) {

    if (!suppressing)
      result << text.substr(lastPos, pos - lastPos);

    lastPos = pos;

    if (pos + 1 < text.length()) {
      if (text[pos + 1] == '$') { // $$ -> $
        if (!suppressing)
          result << '$';

        lastPos += 2;
      } else if (text[pos + 1] == '{') {
        std::size_t startName = pos + 2;
        std::size_t endName = text.find_first_of(" \r\n\t}", startName);

        args.clear();
        std::size_t endVar = parseArgs(text, endName, args);

        if (endVar == std::string::npos) {
          std::stringstream errorStream;
          errorStream << "variable syntax error near \"" << text.substr(pos)
                      << "\"";
          errorText_ = errorStream.str();
          LOG_ERROR(errorText_);
          return false;
        }

        std::string name = text.substr(startName, endName - startName);
        std::size_t nl = name.length();

        if (nl > 2 && name[0] == '<' && name[nl - 1] == '>') {
          if (name[1] != '/') {
            std::string cond = name.substr(1, nl - 2);
            conditions.push_back(cond);
            if (suppressing || !conditionValue(cond))
              ++suppressing;
          } else {
            std::string cond = name.substr(2, nl - 3);
            if (conditions.empty() || conditions.back() != cond) {
              std::stringstream errorStream;
              errorStream << "mismatching condition block end: " << cond;
              errorText_ = errorStream.str();
              LOG_ERROR(errorText_);
              return false;
            }
            conditions.pop_back();

            if (suppressing)
              --suppressing;
          }
        } else {
          if (!suppressing) {
            std::size_t colonPos = name.find(':');

            bool handled = false;
            if (colonPos != std::string::npos) {
              std::string fname = name.substr(0, colonPos);
              std::string arg0 = name.substr(colonPos + 1);
              args.insert(args.begin(), WString::fromUTF8(arg0));
              if (resolveFunction(fname, args, result))
                handled = true;
              else
                args.erase(args.begin());
            }

            if (!handled)
              resolveString(name, args, result);
          }
        }

        lastPos = endVar + 1;
      } else {
        if (!suppressing)
          result << '$'; // $. -> $.
        lastPos += 1;
      }
    } else {
      if (!suppressing)
        result << '$'; // $ at end of template -> $
      lastPos += 1;
    }

    pos = lastPos;
  }

  result << text.substr(lastPos);
  return true;
}

std::size_t WTemplate::parseArgs(const std::string& text,
                                 std::size_t pos,
                                 std::vector<WString>& result)
{
  std::size_t Error = std::string::npos;

  if (pos == std::string::npos)
    return Error;

  enum { Next, Name, Value, SValue, DValue } state = Next;

  WStringStream v;

  for (; pos < text.length(); ++pos) {
    char c = text[pos];
    switch (state) {
    case Next:
      if (!std::isspace(c)) {
        if (c == '}')
          return pos;
        else if (std::isalpha(c) || c == '_') {
          state = Name;
          v.clear();
          v << c;
        } else if (c == '\'') {
          state = SValue;
          v.clear();
        } else if (c == '"') {
          state = DValue;
          v.clear();
        } else
          return Error;
      }
      break;

    case Name:
      if (c == '=') {
        state = Value;
        v << '=';
      } else if (std::isspace(c)) {
        result.push_back(WString::fromUTF8(v.str()));
        state = Next;
      } else if (c == '}') {
        result.push_back(WString::fromUTF8(v.str()));
        return pos;
      } else if (std::isalnum(c) || c == '_' || c == '-' || c == '.')
        v << c;
      else
        return Error;
      break;

    case Value:
      if (c == '\'')
        state = SValue;
      else if (c == '"')
        state = DValue;
      else
        return Error;
      break;

    case SValue:
    case DValue:
      char quote = state == SValue ? '\'' : '"';

      std::size_t end = text.find(quote, pos);
      if (end == std::string::npos)
        return Error;
      if (text[end - 1] == '\\')
        v << text.substr(pos, end - pos - 1) << quote;
      else {
        v << text.substr(pos, end - pos);
        result.push_back(WString::fromUTF8(v.str()));
        state = Next;
      }

      pos = end;
    }
  }

  return pos == text.length() ? std::string::npos : pos;
}

void WTemplate::format(std::ostream& result, const std::string& s,
                       TextFormat textFormat)
{
  format(result, WString::fromUTF8(s), textFormat);
}

void WTemplate::format(std::ostream& result, const WString& s,
                       TextFormat textFormat)
{
  if (textFormat == TextFormat::XHTML) {
    WString v = s;
    if (removeScript(v)) {
      result << v.toUTF8();
      return;
    } else {
      EscapeOStream sout(result);
      sout.append(v.toUTF8(), *plainTextNewLineEscStream_);
      return;
    }
  } else if (textFormat == TextFormat::Plain) {
    EscapeOStream sout(result);
    sout.append(s.toUTF8(), *plainTextNewLineEscStream_);
    return;
  }

  result << s.toUTF8();
}

void WTemplate::propagateRenderOk(bool deep)
{
  changed_ = false;

  WInteractWidget::propagateRenderOk(deep);
}

void WTemplate::enableAjax()
{
  WInteractWidget::enableAjax();
}

DomElementType WTemplate::domElementType() const
{
  DomElementType type = isInline() ? DomElementType::SPAN : DomElementType::DIV;

  WContainerWidget *p = dynamic_cast<WContainerWidget *>(parentWebWidget());
  if (p && p->isList())
    type = DomElementType::LI;

  return type;
}

void WTemplate::setInternalPathEncoding(bool enabled)
{
  if (encodeInternalPaths_ != enabled) {
    encodeInternalPaths_ = enabled;
    changed_ = true;
    repaint();
  }
}

void WTemplate::setEncodeTemplateText(bool on)
{
  encodeTemplateText_ = on;
}

void WTemplate::refresh()
{
  if (text_.refresh() || !strings_.empty()) {
    changed_ = true;
    repaint(RepaintFlag::SizeAffected);
  }

  WInteractWidget::refresh();
}

void WTemplate::reset()
{
  changed_ = true;
  repaint(RepaintFlag::SizeAffected);
}

}
